AI-enhanced audio enhancement